package com.ruoyi.ai;

import com.ruoyi.ai.model.Message;
import lombok.Data;

import java.util.List;

/**
 * @Author: chenss
 * @CreateTime: 2025-02-18
 * @Description:
 * @Version: 1.0
 */

@Data
public class ChatRequest {
    private String model = "deepseek-chat";
    private List<Message> messages;
    private boolean stream = true;
    private int max_tokens = 2048;
    private double temperature = 0.7;
    private double top_p = 0.7;
    private int top_k = 50;
    private double frequency_penalty = 0.5;
    private int n = 1;
    private ResponseFormat response_format = new ResponseFormat("text");

    @Data
    public static class ResponseFormat {
        private String type;

        public ResponseFormat(String type) {
            this.type = type;
        }
    }
}

